// Copyright 2007-2019 Chris Patterson, Dru Sellers, Travis Smith, et. al.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use
// this file except in compliance with the License. You may obtain a copy of the
// License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
// specific language governing permissions and limitations under the License.
namespace MassTransit.AutomatonymousLamarIntegration.Registration
{
    using Automatonymous;
    using Automatonymous.Registration;
    using Microsoft.Extensions.DependencyInjection;
    using Microsoft.Extensions.DependencyInjection.Extensions;


    public class LamarSagaStateMachineRegistrar :
        ISagaStateMachineRegistrar
    {
        readonly IServiceCollection _collection;

        public LamarSagaStateMachineRegistrar(IServiceCollection collection)
        {
            _collection = collection;
        }

        public void RegisterStateMachineSaga<TStateMachine, TInstance>()
            where TStateMachine : class, SagaStateMachine<TInstance>
            where TInstance : class, SagaStateMachineInstance
        {
            _collection.TryAddSingleton<ISagaStateMachineFactory, LamarSagaStateMachineFactory>();
            _collection.TryAddSingleton<IStateMachineActivityFactory, LamarStateMachineActivityFactory>();

            _collection.AddSingleton<TStateMachine>();
            _collection.AddSingleton<SagaStateMachine<TInstance>>(provider => provider.GetRequiredService<TStateMachine>());
        }
    }
}